from langchain_community.utilities import SerpAPIWrapper
import yaml


def get_UID(input: str):
    with open('config.yaml', 'r') as file:
        data = yaml.safe_load(file)

    search = SerpAPIWrapper(serpapi_api_key=data['serpapi_api_key'])
    res = search.run(input)
    return res


# def get_UID():
#     with open('config.yaml', 'r') as file:
#         data = yaml.safe_load(file)

#     search = SerpAPIWrapper(serpapi_api_key=data['serpapi_api_key'])
#     res = search.run("我要怎么从北京飞往上海")
#     return res

